from load_data import load_data
# 声明数据读取函数，从训练集中读取数据
train_loader = load_data('train')
# 以迭代的形式读取数据
for batch_id, data in enumerate(train_loader()):
    image_data, label_data = data
    # if batch_id == 0:
        # 打印数据shape和类型
    print(batch_id,image_data.shape, label_data.shape, type(image_data), type(label_data))
    break